import random
from itertools import permutations


def generate_test_cases():
    # 简单情况
    case1 = {
        "equations": [["a", "b"], ["b", "c"]],
        "values": [2.0, 3.0],
        "queries": [["a", "c"], ["b", "a"], ["a", "e"], ["a", "a"], ["x", "x"]],
    }

    # 复杂情况
    vars = ["x", "y", "z", "w"]
    equations = list(permutations(vars, 2))[:6]
    values = [round(random.uniform(0.1, 5.0), 1) for _ in range(6)]
    queries = list(permutations(vars, 2)) + [["a", "a"], ["x", "q"]]

    case2 = {"equations": equations, "values": values, "queries": queries}

    return [case1, case2]
